Write a decorator with parameter in Python